package cn.javabb.project.entity;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import cn.javabb.common.constant.GenConstants;
import cn.javabb.common.web.BaseEntity;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.*;
import org.apache.commons.lang3.ArrayUtils;

import javax.validation.Valid;
import javax.validation.constraints.NotBlank;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 业务表 gen_table
 *
 */
@Data
@Builder
@ToString
@NoArgsConstructor
@AllArgsConstructor
@ApiModel("生成表")
@TableName("gen_table")
public class GenTable extends BaseEntity {
    /**
     * 编号
     */
    @TableId(value = "table_id")
    @ApiModelProperty("表ID")
    private String tableId;
    /**
     * 数据源主键
     */
    @NotBlank(message = "项目ID不能为空")
    @ApiModelProperty("项目ID")
    private String projectId;
    /**
     * 数据源类型
     */
    @TableField(exist = false)
    @ApiModelProperty("数据源类型")
    private String dbType;
    /**
     * 表名称
     */
    @NotBlank(message = "表名称不能为空")
    @ApiModelProperty(value = "表名")
    private String tableName;
    /**
     * 表描述
     */
    @NotBlank(message = "表描述不能为空")
    @ApiModelProperty(value = "表描述")
    private String tableComment;
    /**
     * 实体类名称(首字母大写)
     */
    @NotBlank(message = "实体类名称不能为空")
    @ApiModelProperty(value = "实体类名称(首字母大写)")
    private String className;
    /**
     * 使用的模板（crud单表操作 tree树表操作 sub主子表操作）
     */
    @ApiModelProperty(value = "使用的模板",notes = "（crud单表操作 tree树表操作 sub主子表操作）")
    private String tplCategory;
    /**
     * 生成包路径
     */
    @ApiModelProperty("生成包路径")
    private String packageName;
    /**
     * 生成模块名
     */
    @ApiModelProperty("生成模块名")
    private String moduleName;
    /**
     * 生成业务名
     */
    @ApiModelProperty("生成业务名")
    private String businessName;
    /**
     * 生成代码方式（0zip压缩包 1自定义路径）
     */
    private String genType;
    /**
     * 生成路径（不填默认项目路径）
     */
    private String genPath;
    /**
     * 关联父表的表名
     */
    private String subTableName;
    /**
     * 本表关联父表的外键名
     */
    private String subTableFkName;
    /**
     * 主键信息
     */
    @TableField(exist = false)
    private GenTableColumn pkColumn;
    /**
     * 子表信息
     */
    @TableField(exist = false)
    private GenTable subTable;
    /**
     * 表列信息
     */
    @Valid
    @TableField(exist = false)
    private List<GenTableColumn> columns;
    /**
     * 其它生成选项
     */
    @ApiModelProperty("其它生成选项")
    private String options;
    @ApiModelProperty("备注")
    private String remark;

    /**
     * 树编码字段
     */
    @TableField(exist = false)
    private String treeCode;

    /**
     * 树父编码字段
     */
    @TableField(exist = false)
    private String treeParentCode;

    /**
     * 树名称字段
     */
    @TableField(exist = false)
    private String treeName;

    /**
     * 上级菜单ID字段
     */
    @TableField(exist = false)
    private String parentMenuId;

    /**
     * 上级菜单名称字段
     */
    @TableField(exist = false)
    private String parentMenuName;

    /*public String getPackageName() {
        if (StrUtil.isNotBlank(this.moduleName)) {
            return packageName + "." + this.moduleName;
        }
        return packageName;
    }*/

    public GenTable(String tableId) {
        this.tableId = tableId;
    }

    public boolean isSub() {
        return isSub(this.tplCategory);
    }

    public static boolean isSub(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_SUB, tplCategory);
    }

    public boolean isTree() {
        return isTree(this.tplCategory);
    }

    public static boolean isTree(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_TREE, tplCategory);
    }

    public boolean isCrud() {
        return isCrud(this.tplCategory);
    }

    public static boolean isCrud(String tplCategory) {
        return tplCategory != null && StrUtil.equals(GenConstants.TPL_CRUD, tplCategory);
    }

    public boolean isSuperColumn(String javaField) {
        return isSuperColumn(this.tplCategory, javaField);
    }

    public static boolean isSuperColumn(String tplCategory, String javaField) {
        if (isTree(tplCategory)) {
            return StrUtil.equalsAnyIgnoreCase(javaField,
                    ArrayUtils.addAll(GenConstants.TREE_ENTITY, GenConstants.BASE_ENTITY));
        }
        return StrUtil.equalsAnyIgnoreCase(javaField, GenConstants.BASE_ENTITY);
    }
}